package com.starhub.application.model.enums;

public enum ScoringModelTypeEnum {
    BCEV1("bce-reranker-base_v1", ChatProviderEnums.Baidu, "百度bce-reranker-base_v1"),
    Custom("custom", ChatProviderEnums.Custom, "私有化重排序模型"),
    // 可以继续添加其他厂商的向量模型
    ;

    private final String code;
    private final ChatProviderEnums chatProvider;
    private final String desc;

    ScoringModelTypeEnum(String code, ChatProviderEnums chatProvider, String desc) {
        this.code = code;
        this.chatProvider = chatProvider;
        this.desc = desc;
    }

    public String getCode() {
        return code;
    }

    public ChatProviderEnums getChatProvider() {    
        return chatProvider;
    }

    public String getDesc() {
        return desc;
    }


    public static ScoringModelTypeEnum getByCode(String code) {
        for (ScoringModelTypeEnum type : values()) {
            if (type.getCode().equals(code)) {
                return type;
            }
        }
        return null;
    }
} 
